How to unit test a method that contains a database call in Python